from artifacts.base.attr_type import AttrType
from artifacts.base.sr.base_relics import Relics


class Head(Relics):
    def get_name(self):
        return '头'

    def get_main_attr_weight_map(self):
        return {
            AttrType.SR_HP: 100
        }


class Hands(Relics):
    def get_name(self):
        return '手'

    def get_main_attr_weight_map(self):
        return {
            AttrType.SR_ATK: 100
        }


class Body(Relics):
    def get_name(self):
        return '衣服'

    def get_main_attr_weight_map(self):
        return {
            AttrType.SR_HP_P: 2000,
            AttrType.SR_ATK_P: 2000,
            AttrType.SR_DEF_P: 2000,
            AttrType.SR_EFFECT_RATE: 1000,
            AttrType.SR_HEALING: 1000,
            AttrType.SR_CRIT_RATE: 1000,
            AttrType.SR_CRIT_DMG: 1000,
        }


class Feet(Relics):
    def get_name(self):
        return '鞋'

    def get_main_attr_weight_map(self):
        return {
            AttrType.SR_HP_P: 3000,
            AttrType.SR_ATK_P: 3000,
            AttrType.SR_DEF_P: 3000,
            AttrType.SR_SPD: 1000
        }


class Link(Relics):
    def get_name(self):
        return '链子'

    def get_main_attr_weight_map(self):
        return {
            AttrType.SR_HP_P: 2700,
            AttrType.SR_ATK_P: 2700,
            AttrType.SR_DEF_P: 2400,
            AttrType.SR_BREAK_EFFECT: 1600,
            AttrType.SR_ENERGY: 600,
        }


class Sphere(Relics):
    def get_name(self):
        return '球'

    def get_main_attr_weight_map(self):
        return {
            AttrType.SR_HP_P: 1200,
            AttrType.SR_ATK_P: 1200,
            AttrType.SR_DEF_P: 1200,
            AttrType.SR_QUANTUM_DMG: 900,
            AttrType.SR_IMAGINARY_DMG: 900,
            AttrType.SR_LIGHTING_DMG: 900,
            AttrType.SR_FIRE_DMG: 900,
            AttrType.SR_WIND_DMG: 900,
            AttrType.SR_ICE_DMG: 900,
            AttrType.SR_PHYSICAL_DMG: 900,
        }

